# -*- coding: utf-8 -*-
# 说明：验证 ChromaDB 是否安装成功

# 说明：导入 chromadb 库
import chromadb

# 说明：创建一个临时的内存客户端（不会保存到磁盘）
client = chromadb.EphemeralClient()

# 说明：创建一个测试集合
collection = client.create_collection(name="test")

# 说明：添加一条测试数据
collection.add(
    documents=["这是一个测试", "这是一台全新的笔记本电脑"], ids=["test_1", "2"]
)

# 说明：查询数据，验证功能是否正常
results = collection.query(query_texts=["非常新的电脑"], n_results=1)

# 说明：打印查询结果
print("安装成功！查询结果：", results["documents"])

# 说明：如果没有报错并输出了查询结果，说明安装成功
